home *** CD-ROM | disk | FTP | other *** search
/ Hardcore Visual Basic 5.0 (2nd Edition) / Hardcore Visual Basic 5.0 - Second Edition (1997)(Microsoft Press).iso / Code / REGNOD~1.CLS < prev    next >
Text File  |  1997-06-14  |  3KB  |  103 lines
  1. VERSION 1.0 CLASS
  2. BEGIN
  3.   MultiUse = -1  'True
  4. END
  5. Attribute VB_Name = "CRegNodeWalker"
  6. Attribute VB_GlobalNameSpace = False
  7. Attribute VB_Creatable = True
  8. Attribute VB_PredeclaredId = False
  9. Attribute VB_Exposed = True
  10. Option Explicit
  11.  
  12. Public Enum EErrorRegNodeWalker
  13.     eeBaseRegNodeWalker = 13190     ' CRegNodeWalker
  14. End Enum
  15.  
  16. ' Implement Basic-friendly version of IEnumVARIANT
  17. Implements IVariantWalker
  18. ' Delegate to class that implements real IEnumVARIANT
  19. Private vars As CEnumVariant
  20. ' Connect back to parent collection
  21. Private connect As CRegNode
  22.  
  23. ' Private state data
  24. Private iCur As Long
  25.  
  26. Private Sub Class_Initialize()
  27.     ' Initialize position in collection
  28.     iCur = -1
  29.     ' Connect walker to CEnumVariant so it can call methods
  30.     Set vars = New CEnumVariant
  31.     vars.Attach Me
  32. End Sub
  33.  
  34. ' Receive connection from CRegNode
  35. Sub Attach(connectA As CRegNode)
  36.     Set connect = connectA
  37. End Sub
  38.  
  39. ' Return IEnumVARIANT (indirectly) to client collection
  40. Friend Function NewEnum() As stdole.IEnumVARIANT
  41.     Set NewEnum = vars
  42. End Function
  43.  
  44. ' Implement IVariantWalker methods
  45. Private Function IVariantWalker_More(v As Variant) As Boolean
  46.     ' We can't fail in a walker, and yet the registry will return errors
  47.     On Error Resume Next
  48.     iCur = iCur + 1
  49.     Do While iCur < connect.NodeCount
  50.         ' Get next node from registry
  51.         Set v = connect.Nodes(iCur)
  52.         Select Case Err.Number
  53.         Case 0
  54.             ' If more data, return True
  55.             IVariantWalker_More = True
  56.             Exit Function
  57.         Case ApiToCom(ERROR_ACCESS_DENIED)
  58.             ' WinNT returns access error if you don't have permission
  59.             Err.Clear
  60.             iCur = iCur + 1
  61.         Case ApiToCom(ERROR_NO_MORE_ITEMS)
  62.             ' Sometimes happens under WinNT
  63.             iCur = connect.NodeCount
  64.             Err.Clear
  65.             Exit Function
  66.         Case Else
  67.             ' Failure terminates loop, so do only for catastrophies
  68.             Err.Raise Err.Number, Err.Source, Err.Description
  69.         End Select
  70.     Loop
  71. End Function
  72.  
  73. Private Sub IVariantWalker_Reset()
  74.     ' Move to first element
  75.     iCur = 0
  76. End Sub
  77.  
  78. Private Sub IVariantWalker_Skip(c As Long)
  79.     ' Skip a given number of elements
  80.     iCur = iCur + c
  81. End Sub
  82.  
  83. #If fComponent = 0 Then
  84. Private Sub ErrRaise(e As Long)
  85.     Dim sText As String, sSource As String
  86.     If e > 1000 Then
  87.         sSource = App.ExeName & ".RegNodeWalker"
  88.         Select Case e
  89.         Case eeBaseRegNodeWalker
  90.             BugAssert True
  91.        ' Case ee...
  92.        '     Add additional errors
  93.         End Select
  94.         Err.Raise COMError(e), sSource, sText
  95.     Else
  96.         ' Raise standard Visual Basic error
  97.         sSource = App.ExeName & ".VBError"
  98.         Err.Raise e, sSource
  99.     End If
  100. End Sub
  101. #End If
  102.  
  103.